package com.ai.aiagent.factory;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;
//如果不使用自定义处理器，或者未启用‌ “允许空上下文” 选项，系统在找不到相关文档时会默认改写用户查询‌ userText：
//The user query is outside your knowledge base.
//Politely inform the user that you can't answer it.
//自定义处理器,当系统无法找到相关文档时，会返回我们自定义的友好提示：
public class LoveAppContextualQueryAugmenterFactory {
    public static ContextualQueryAugmenter createInstance() {
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉，我只能回答恋爱相关的问题，别的没办法帮到您哦，
                有问题可以联系客服 2310053976@qq.com
                """);
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false)
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
